package com.example.mq.mqserver.datacenter;

import com.example.mq.MqApplication;
import com.example.mq.mqserver.core.Binding;
import com.example.mq.mqserver.core.Exchange;
import com.example.mq.mqserver.core.ExchangeType;
import com.example.mq.mqserver.core.MSGQueue;
import com.example.mq.mqserver.mapper.MetaMapper;

import javax.annotation.Resource;
import java.io.File;
import java.util.List;

/**
 * Created with IntelliJ IDEA.
 *
 * @Author: DongGuoZhen
 * @Date: 2024/02/23/19:43
 * @Description: 通过这个类整合上述的数据库操作
 */
public class DataBaseManager {

    //从Spring中拿到现成对象
    private MetaMapper metaMapper;

    //针对数据库进行初始化
    public void init() {
        //手动获取到MetaMapper对象
        metaMapper = MqApplication.context.getBean(MetaMapper.class);

        //建表 插入默认数据的操作
        //希望在程序启动的时候,如果数据库已经存在,或者表都有了,就不做操作
        //如果不存在,就创建库,构造默认数据
        if(!checkDBExists()) {
            //如果数据库不存在,我们就建库建表
            //先创建一个data目录
            File dataDir = new File("./data");
            dataDir.mkdirs();
            //创建数据表
            createTable();
            //插入默认数据
            createDefaultData();
            System.out.println("[DataBaseManager] 数据库初始化完成!");
        } else {
            //数据库已经存在
            System.out.println("[DataBaseManager] 数据库已经存在!");
        }
    }

    /**
     * 删除数据库文件和目录
     */
    public void deleteDB() {
        File file = new File("./data/meta.db");
        boolean ret = file.delete();
        if(ret) {
            System.out.println("[DataBaseManager] 删除数据库文件成功");
        } else {
            System.out.println("[DataBaseManager] 删除数据库文件失败");
        }

        File dataDir = new File("./data");
        //使用delete删除目录的时候,需要保证目录是空的
        ret = dataDir.delete();
        if(ret) {
            System.out.println("[DataBaseManager] 删除数据库目录成功");
        } else {
            System.out.println("[DataBaseManager] 删除数据库目录失败");
        }
    }

    /**
     * 判定数据库是否存在  判断meta.db文件是否存在
     * @return
     */
    private boolean checkDBExists() {
        File file = new File("./data/meta.db");
        return file.exists();
    }

    /**
     * 这个方法用来建表 建库操作不需要手动执行(不需要创建meta.db这个文件)
     * 在首次执行这里的数据库操作的时候,会自动的创建出meta.db,  MyBatis自动帮我们完成
     */
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager] 创建表完成!");
    }

    /**
     * 给数据库表中添加默认数据
     * 主要是添加默认的交换机
     * RabbitMQ中有这样的设定,带有一个匿名交换机,类型为直接交换机(DIRECT)
     */
    private void createDefaultData() {
        //构造一个默认交换机
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        //向数据库表中添加一个交换机
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager] 创建初始数据完成!");
    }



    //把其他的数据库操作也在这类中封装一下
    public void insertExchange(Exchange exchange) {
        metaMapper.insertExchange(exchange);
    }

    public void deleteExchange(String exchangeName) {
        metaMapper.deleteExchange(exchangeName);
    }

    public List<Exchange> selectAllExchanges() {
        return metaMapper.selectAllExchanges();
    }

    public void insertQueue(MSGQueue msgQueue) {
        metaMapper.insertQueue(msgQueue);
    }

    public void deleteQueue(String queueName) {
        metaMapper.deleteQueue(queueName);
    }

    public List<MSGQueue> selectAllQueues() {
        return metaMapper.selectAllQueues();
    }

    public void insertBinding(Binding binding) {
        metaMapper.insertBinding(binding);
    }

    public void deleteBinding(Binding binding) {
        metaMapper.deleteBinding(binding);
    }

    public List<Binding> selectBindings() {
        return metaMapper.selectAllBindings();
    }

}